#pragma once

#ifdef __cplusplus
extern "C" {
#endif

  int gemm_cutlass(
    int M, int N, int K,
    void *A, int stride_A,
    void *B, int stride_B,
    void *C, int stride_C,
    void *D, int stride_D,
    const float alpha, const float beta
  );

#ifdef __cplusplus
}
#endif
